What We Keep of What We Forget: Replacement Operators and Routing Semantics in Training-Free MoE Expert Pruning
训练无关 MoE 专家剪枝中的替换算子与路由语义
🧭 MoE 专家剪枝 · 替换算子与路由语义 ★ 深度解读
School of Electrical Engineering and Computer Science (SEECS), National University of Sciences and Technology (NUST) · Deep Learning Laboratory, National Center of Artificial Intelligence (NCAI), NUST
固定剪枝掩码,系统比较 MoE 专家替换算子与门控尺度,提出倾斜新手算子并揭示 trained-scale rule
📌 工作参考价值(4 点)
- 部署 MoE 推荐/Agent 模型时,剪枝后不要默认 renormalize 幸存者 gate;先检查原模型 router 是否在训练时做了 top-k renormalization,保留与其一致的 gate scale(trained-scale rule),否则可能掉 0.8–8.5 个点,甚至反转 rerouting 是否优于均值常数。
- 若要对剪掉的专家做廉价替代,可用 tilted novice:只存一个低秩输入映射(rank-32 甚至 rank-8),参数增加约 1.2–1.8%,在多项选择和生成任务上稳定优于均值常数,且对 3/4-bit 量化鲁棒;比保留更多专家更省参数。
- 评估剪枝模型不能只看 MCQ;务必加入 GSM8K、代码生成等推理/生成任务,因为 25% 剪枝时 MCQ 可保留 92–97%,但 GSM8K 仅 5–65%、代码 0–17%,结论可能反转。
- 校准数据域对剪枝 mask 影响很大(4–9 倍于重采样),推荐/Agent 场景应使用与线上分布一致的校准数据,并报告不同校准切片的结果;仅用通用文本可能掩盖领域退化。